{
  "color_mnist": {
    "model_path": "./checkpoints/standard_supervised/color_mnist",
    "num_models": 5,
    "label_budget": 500,
    "batch_size": 128,
    "sampling_rounds": 10,
    "max_epochs": 200,
    "patience_epochs": 10,
    "min_epochs": 50,
    "optimizer_name": "Adam",
    "optimizer_kargs": {
      "learning_rate": 1e-3
    },
    "self_train_kwargs": {
      "pseudo_train_epochs": 20,
      "pseudo_ckpt_epoch": 5,
      "frac": 0.1,
      "use_checkpoint_ensemble": true,
      "lower_threshold": 0.9,
      "upper_threshold": 1.0
    },
    "finetune_method": "joint_train",
    "finetune_kwargs": {
      "ckpt_epoch": 5,
      "lambda": 1.0,
      "init_steps": 1000,
      "init_ckpt_step": 200
    },
    "debug_info": false,
    "print_freq": 500
  },
  "cifar10": {
    "model_path": "./checkpoints/standard_supervised/cifar10",
    "num_models": 5,
    "label_budget": 500,
    "batch_size": 128,
    "sampling_rounds": 10,
    "max_epochs": 200,
    "patience_epochs": 10,
    "min_epochs": 50,
    "optimizer_name": "SGD",
    "optimizer_kargs": {
      "learning_rate": 5e-3,
      "momentum": 0.9
    },
    "self_train_kwargs": {
      "pseudo_train_epochs": 20,
      "pseudo_ckpt_epoch": 5,
      "frac": 0.1,
      "use_checkpoint_ensemble": true,
      "lower_threshold": 0.9,
      "upper_threshold": 1.0
    },
    "finetune_method": "joint_train",
    "finetune_kwargs": {
      "ckpt_epoch": 5,
      "lambda": 1.0,
      "init_steps": 1000,
      "init_ckpt_step": 200
    },
    "debug_info": false,
    "print_freq": 500
  },
  "domainnet": {
    "model_path": "./checkpoints/standard_supervised/domainnet",
    "num_models": 5,
    "label_budget": 500,
    "batch_size": 128,
    "sampling_rounds": 10,
    "max_epochs": 200,
    "patience_epochs": 10,
    "min_epochs": 50,
    "optimizer_name": "Adam",
    "optimizer_kargs": {
      "learning_rate": 1e-4
    },
    "self_train_kwargs": {
      "pseudo_train_epochs": 20,
      "pseudo_ckpt_epoch": 5,
      "frac": 0.1,
      "use_checkpoint_ensemble": true,
      "lower_threshold": 0.9,
      "upper_threshold": 1.0
    },
    "finetune_method": "joint_train",
    "finetune_kwargs": {
      "ckpt_epoch": 5,
      "lambda": 1.0,
      "init_steps": 1000,
      "init_ckpt_step": 200
    },
    "debug_info": false,
    "print_freq": 500
  },
  "fmow": {
    "model_path": "./checkpoints/standard_supervised/fmow",
    "num_models": 5,
    "label_budget": 500,
    "batch_size": 128,
    "sampling_rounds": 10,
    "max_epochs": 200,
    "patience_epochs": 10,
    "min_epochs": 50,
    "optimizer_name": "Adam",
    "optimizer_kargs": {
      "learning_rate": 1e-4
    },
    "self_train_kwargs": {
      "pseudo_train_epochs": 20,
      "pseudo_ckpt_epoch": 5,
      "frac": 0.1,
      "use_checkpoint_ensemble": true,
      "lower_threshold": 0.9,
      "upper_threshold": 1.0
    },
    "finetune_method": "joint_train",
    "finetune_kwargs": {
      "ckpt_epoch": 5,
      "lambda": 1.0,
      "init_steps": 1000,
      "init_ckpt_step": 200
    },
    "debug_info": false,
    "print_freq": 500
  },
  "amazon_review": {
    "model_path": "./checkpoints/standard_supervised/amazon_review",
    "num_models": 5,
    "label_budget": 500,
    "batch_size": 128,
    "sampling_rounds": 10,
    "max_epochs": 200,
    "patience_epochs": 10,
    "min_epochs": 50,
    "optimizer_name": "Adam",
    "optimizer_kargs": {
      "learning_rate": 1e-3
    },
    "self_train_kwargs": {
      "pseudo_train_epochs": 20,
      "pseudo_ckpt_epoch": 5,
      "frac": 0.1,
      "use_checkpoint_ensemble": true,
      "lower_threshold": 0.9,
      "upper_threshold": 1.0
    },
    "finetune_method": "joint_train",
    "finetune_kwargs": {
      "ckpt_epoch": 5,
      "lambda": 1.0,
      "init_steps": 1000,
      "init_ckpt_step": 200
    },
    "debug_info": false,
    "print_freq": 500
  },
  "otto": {
    "model_path": "./checkpoints/standard_supervised/otto",
    "num_models": 5,
    "label_budget": 500,
    "batch_size": 128,
    "sampling_rounds": 10,
    "max_epochs": 200,
    "patience_epochs": 10,
    "min_epochs": 50,
    "optimizer_name": "Adam",
    "optimizer_kargs": {
      "learning_rate": 1e-3
    },
    "self_train_kwargs": {
      "pseudo_train_epochs": 20,
      "pseudo_ckpt_epoch": 5,
      "frac": 0.1,
      "use_checkpoint_ensemble": true,
      "lower_threshold": 0.9,
      "upper_threshold": 1.0
    },
    "finetune_method": "joint_train",
    "finetune_kwargs": {
      "ckpt_epoch": 5,
      "lambda": 1.0,
      "init_steps": 1000,
      "init_ckpt_step": 200
    },
    "debug_info": false,
    "print_freq": 500
  }
}